package com.yh.jandan.ext;

import javax.sql.DataSource;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.Sqlite3Dialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.c3p0.C3p0Plugin;
import com.jfinal.plugin.druid.DruidPlugin;

/**
 * GeneratorDemo
 */   
public class GeneratorDemo {
	  
	public static Prop getProp() {
		return PropKit.use( "a_little_config.txt" );
	}
	
	public static DataSource getDataSourceC3p0() {
		Prop p = getProp();
		C3p0Plugin c3p0Plugin = new C3p0Plugin( p.get( "jdbcUrl" ) , p.get( "user" ) , p.get( "password" ) );
		c3p0Plugin.start();
		return c3p0Plugin.getDataSource();
	}
	
	public static DataSource getDataSourceDruid() {
		Prop p = GeneratorDemo.getProp();
		DruidPlugin dp = new DruidPlugin( p.get( "jdbcUrl" ) , p.get( "user" ) , p.get( "password" ) );
		// dp.addFilter( new StatFilter() );
		// WallFilter wall = new WallFilter();
		// wall.setDbType( "mysql" );
		// dp.addFilter( wall );
		dp.start();
		return dp.getDataSource();
	}
	
	public static DataSource getDataSource() {
		// return getDataSourceC3p0();
		return getDataSourceDruid();
	}
	
	private static String getPackagePath( String baseModelPackageName ) {
		return baseModelPackageName.replace( "." , "/" );
	}
	
	public static void main( String[] args ) {
		// base model 所使用的包名
		String baseModelPackageName = "com.yh.jandan.ext.jfinal";
		// base model 文件保存路径
		String baseModelOutputDir = PathKit.getRootClassPath() + "/../src/" + getPackagePath( baseModelPackageName );
		// String baseModelOutputDir = "src/" + getPackagePath( baseModelPackageName );
		
		// System.out.println( baseModelOutputDir );
		
		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = "com.yh.jandan.model";
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = PathKit.getRootClassPath() + "/../src/" + getPackagePath( modelPackageName );
		
		// 创建生成器
		Generator gernerator = new Generator( getDataSource() , baseModelPackageName , baseModelOutputDir , modelPackageName , modelOutputDir );
		// 设置数据库方言
		gernerator.setDialect( new Sqlite3Dialect() );
		// 添加不需要生成的表名
		gernerator.addExcludedTable( "sqlite_sequence" );
		// 设置是否在 Model 中生成 dao 对象
		gernerator.setGenerateDaoInModel( true );
		// 设置是否生成字典文件
		gernerator.setGenerateDataDictionary( false );
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
		// gernerator.setRemovedTableNamePrefixes( "t_" );
		// 生成
		gernerator.generate();
	}
	
}
